from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope import snapshot_download

# # 模型下载地址
# model_dir = snapshot_download("damo/nlp_deberta_rex-uninlu_chinese-base", revision="v1.2.1")
# print(model_dir)

# # 加载模型
# semantic_cls = pipeline('rex-uninlu', model=model_dir)

# # 输入文本
# text = "你好，我想订一张去北京的机票"

# # 调用模型进行自然语言理解
# result = nlu(text)

# # 输出结果
# print(result)

semantic_cls = pipeline('rex-uninlu', model='damo/nlp_deberta_rex-uninlu_chinese-base', model_revision='v1.2.1')

# 命名实体识别 {实体类型: None}
semantic_cls(
    input='1944年毕业于北大的名古屋铁道会长谷口清太郎等人在日本积极筹资，共筹款2.7亿日元，参加捐款的日本企业有69家。', 
    schema={
        '人物': None,
        '地理位置': None,
        '组织机构': None
    }
) 